package com.atguigu.aiproject.utils;

/**
 * ThreadLocal工具类，用于保存当前线程的用户信息
 */
public class UserContext {

    //注意每个ThreadLocal实例是不同的，不能用同一个ThreadLocal实例来保存多个字段
    // 存储用户名
    private static final ThreadLocal<String> usernameThreadLocal = new ThreadLocal<>();

    // 存储用户ID
    private static final ThreadLocal<Integer> userIdThreadLocal = new ThreadLocal<>();

    // 可以根据需要添加更多用户信息字段
    // private static final ThreadLocal<Long> userIdThreadLocal = new ThreadLocal<>();

    /**
     * 设置用户名
     */
    public static void setUsername(String username) {
        usernameThreadLocal.set(username);
    }
    /**
     * 设置用户ID
     */
    public static void setUserId(int userId) {
        userIdThreadLocal.set(userId);
    }

    /**
     * 获取用户名
     */
    public static String getUsername() {
        return usernameThreadLocal.get();
    }
    public static Integer getUserId() {
        return userIdThreadLocal.get();
    }

    /**
     * 清除所有ThreadLocal数据
     */
    public static void clear() {
        usernameThreadLocal.remove();
        userIdThreadLocal.remove();
        // 如果有其他字段也需要清除
        // userIdThreadLocal.remove();
    }

    // 可以根据需要添加更多getter和setter方法
    /*
    public static void setUserId(Long userId) {
        userIdThreadLocal.set(userId);
    }

    public static Long getUserId() {
        return userIdThreadLocal.get();
    }
    */
}

